import api from "@/util/api";

// 简单聊天接口
export function chatWithAI(message) {
  return api.post("/app/chat/ask", message).then((resp) => resp.data);
}

// 高级聊天接口，支持更多参数
export function advancedChatWithAI(data) {
  return api.post("/app/chat/chat", data).then((resp) => resp.data);
}

// 安全模式聊天接口
export function safeChatWithAI(message, model) {
  const url = `/app/chat/ask-safe${model ? `?model=${model}` : ""}`;
  return api.post(url, message).then((resp) => resp.data);
}

// 获取支持的模型列表
export function getAIModels() {
  return api.get("/app/chat/models").then((resp) => resp.data);
}

// 获取欢迎消息
export function getWelcomeMessage() {
  return api.get("/app/chat/welcome").then((resp) => resp.data.message);
}

const zhipuAIApi = {
  chatWithAI,
  advancedChatWithAI,
  safeChatWithAI,
  getAIModels,
  getWelcomeMessage,
};

export default zhipuAIApi;
